package com.atguigu.tingshu.user.service.impl;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.IdUtil;
import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.rabbit.constant.MqConst;
import com.atguigu.tingshu.common.rabbit.service.RabbitService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private RabbitService rabbitService;


    /**
     * 根据用户ID及声音ID查询播放进度
     *
     * @param userId
     * @param trackId
     * @return
     */
    @Override
    public BigDecimal getTrackBreakSecond(Long userId, Long trackId) {
        //1.创建查询对象：封装查询条件
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId).and("trackId").is(trackId));
        //2.执行查询：注意每个用户都有自己播放进度集合
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (listenProcess != null) {
            return listenProcess.getBreakSecond();
        }
        return new BigDecimal("0.00");
    }

    /**
     * 保存更新声音播放进度
     *
     * @param userListenProcessVo
     * @return
     */
    @Override
    public void updateListenProcess(Long userId, UserListenProcessVo userListenProcessVo) {
        //1.查询当前用户某个声音播放进度
        //1.1 构建查询条件
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId).and("trackId").is(userListenProcessVo.getTrackId()));
        //1.2 执行查询：注意每个用户都有自己播放进度集合
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        //2.如果不存在构建声音播放进度保存到MongoDB中
        BigDecimal breakSecond = userListenProcessVo.getBreakSecond().setScale(0, RoundingMode.HALF_UP);
        if (listenProcess == null) {
            listenProcess = new UserListenProcess();
            listenProcess.setUserId(userId);
            listenProcess.setTrackId(userListenProcessVo.getTrackId());
            listenProcess.setAlbumId(userListenProcessVo.getAlbumId());
            listenProcess.setBreakSecond(breakSecond);
            listenProcess.setCreateTime(new Date());
            listenProcess.setUpdateTime(new Date());
        } else {
            //3.如果已存在播放进度，更新播放进度时间及更新时间 包含文档注解ID
            listenProcess.setBreakSecond(breakSecond);
            listenProcess.setUpdateTime(new Date());
        }
        mongoTemplate.save(listenProcess, collectionName);

        //4.采用MQ异步更新数据库及索引库中统计信息（播放量）

        //4.1 确保某个用户当日内只能更新一次播放统计量-生产者幂等性（由于播放进度更新会定时调用）
        //4.1.1 构建生产者幂等性Key 形式：前缀:userId_albumId_trackId
        String key = RedisConstant.USER_TRACK_REPEAT_STAT_PREFIX + userId + "_" + userListenProcessVo.getAlbumId() + "_" + userListenProcessVo.getTrackId();
        //4.1.2 计算过期时间 要求：当前日内一次统计有效
        long ttl = DateUtil.endOfDay(new Date()).getTime() - System.currentTimeMillis();
        Boolean flag = redisTemplate.opsForValue().setIfAbsent(key, "", ttl, TimeUnit.MILLISECONDS);
        //4.2 发送更新声音统计VO消息到RabbitMQ
        if (flag) {
            //4.2.1 创建更新统“计消息VO对象
            TrackStatMqVo mqVo = new TrackStatMqVo();
            mqVo.setAlbumId(userListenProcessVo.getAlbumId());
            mqVo.setTrackId(userListenProcessVo.getTrackId());
            mqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            mqVo.setCount(1);
            mqVo.setBusinessNo(IdUtil.randomUUID());
            //4.2.2 发送MQ消息
            rabbitService.sendMessage(MqConst.EXCHANGE_TRACK, MqConst.ROUTING_TRACK_STAT_UPDATE, mqVo);
        }
    }

    /**
     * 获取当前用户最近播放声音
     *
     * @return {albumId:1,trackId:12}
     */
    @Override
    public Map<String, Long> getLatelyTrack(Long userId) {
        //1.查询当前用户某个声音播放进度
        //1.1 构建查询条件
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId));
        query.with(PageRequest.of(0, 1, Sort.Direction.DESC, "updateTime"));
        //1.2 执行查询：注意每个用户都有自己播放进度集合
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
        UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (listenProcess != null) {
            Map<String, Long> map = new HashMap<>();
            map.put("albumId", listenProcess.getAlbumId());
            map.put("trackId", listenProcess.getTrackId());
            return map;
        }
        return null;
    }


}
